You will also have to record rendering in order to be sensitive to what mode you are in:
if( (WebPartManager.DisplayMode == WebPartManager.DesignDisplayMode) ||
(WebPartManager.DisplayMode == WebPartManager.EditDisplayMode) ) {
} else {
}
source
share