Fixed issue with method parameter