Fixing cleanup script.

Signed-off-by: Simarpreet Singh <simar@linux.com>