Fix: bug introduced in previus commit