Removing die() statement