diff --git a/boban.zsh-theme b/boban.zsh-theme
index 3195b11..84da796 100644
--- a/boban.zsh-theme
+++ b/boban.zsh-theme
@@ -148,7 +148,15 @@ prompt_dir() {
 prompt_virtualenv() {
 	local virtualenv_path="$VIRTUAL_ENV"
 	if [[ -n $virtualenv_path && -n $VIRTUAL_ENV_DISABLE_PROMPT ]]; then
-		prompt_segment blue black "(`basename $virtualenv_path`)"
+		prompt_segment blue black ""
+	fi
+}
+
+# AWS profile
+prompt_aws() {
+	local aws_profile="$AWS_PROFILE"
+	if [[ -n $aws_profile ]]; then
+		prompt_segment blue yellow "($aws_profile)"
 	fi
 }
 
@@ -188,6 +196,7 @@ build_prompt() {
 	#RETVAL=$?
 	#prompt_status
 	prompt_virtualenv
+	prompt_aws
 	prompt_context
 	prompt_dir
 	# prompt_tf