Little code fix to the method.