fix for rendering different stages