Adding missing part to the last commit