Fix check if that should be top level when writing

Fix check if that should be top level when writing #588