bugfix correctly referncing the states module