improvement in the last commit

Signed-off-by: Aniket Anvit <seeanvit@gmail.com>