Fix

Lost some code when wandered between c9 and my machine