Minor change to method!