little bit of cleanup, fix error