Will fix this in both methods.