Fixing regression and added test