Fix coding style with returning Unit