minor
fixing two stage, think this will work