Add simple wait state test
Fix typo on return type