add stage tests