Refactored into 1 test