little corection for proper implementation