Fix false positive to close #1

Please test and make sure it's still working as I have moved things into a module