Fixed the remove media feature