Fix for #193

Should no longer be causing problems in the stage edit page